﻿// Gateway.Api 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
// 
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
// 
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

//using Aurora.AI.AzureOpenAI.Extensions;
using Aurora.AI.Claudia.Extensions;
using Aurora.AI.Domain;
using Aurora.AI.ErnieBot.Extensions;
using Aurora.AI.Gateway.Services;
using Aurora.AI.Gateway.Services.ChatChannel;
using Aurora.AI.Hunyuan.Extensions;
using Aurora.AI.MetaGLM.Extensions;
using Aurora.AI.Moonshot.Extensions;
using Aurora.AI.Ollama.Extensions;
using Aurora.AI.OpenAI.Extensions;
using Aurora.AI.Qiansail.Extensions;
using Aurora.AI.SparkDesk.Extensions;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;

namespace Aurora.AI.Gateway;

public static class AiServiceExtension
{
    /// <summary>
    /// 注入服务 初始化表
    /// </summary>
    /// <param name="services"></param>
    /// <param name="configuration"></param>
    public static void AddGatewayAiService(this IServiceCollection services, IConfiguration configuration)
    {
        // add json
        IConfigurationSection section = configuration.GetSection("AI.Gateway.Options");
        string connection = section["ConnectionStrings"]!;
        services.AddDomainService(connection);

        // services.AddAzureOpenAIService()
        services.AddClaudiaService()
            .AddErnieBotService()
            .AddHunyuanService()
            .AddMetaGLMService()
            .AddMoonshotService()
            .AddOllamaService()
            .AddOpenAIService()
            .AddQiansailService()
            .AddSparkDeskService();
        services.AddScoped<CommonService>();
        services.AddScoped<ChatChannelService>();
        services.AddScoped<ImageService>();
        services.AddScoped<RateLimitModelService>();
        services.AddScoped<SettingService>();
        services.AddScoped<UserAILimitService>();
        services.AddScoped<TokenService>();
        services.AddScoped<ChatLoggerService>();
        services.AddSingleton<ChatLoggerEventBus>();
        services.AddScoped<ChannelTestService>();
    }

}